# -*- coding: utf-8 -*-

"""剑指 Offer II 034. 外星语言是否排序
某种外星语也使用英文小写字母，但可能顺序 order 不同。字母表的顺序（order）是一些小写字母的排列。
给定一组用外星语书写的单词 words，以及其字母表的顺序 order，只有当给定的单词在这种外星语中按字典序排列时，返回 true；否则，返回 false。

示例 1：
输入：words = ["hello","leetcode"], order = "hlabcdefgijkmnopqrstuvwxyz"
输出：true
解释：在该语言的字母表中，'h' 位于 'l' 之前，所以单词序列是按字典序排列的。

示例 2：
输入：words = ["word","world","row"], order = "worldabcefghijkmnpqstuvxyz"
输出：false
解释：在该语言的字母表中，'d' 位于 'l' 之后，那么 words[0] > words[1]，因此单词序列不是按字典序排列的。

示例 3：
输入：words = ["apple","app"], order = "abcdefghijklmnopqrstuvwxyz"
输出：false
解释：当前三个字符 "app" 匹配时，第二个字符串相对短一些，然后根据词典编纂规则 "apple" > "app"，因为 'l' > '∅'，其中 '∅' 是空白字符，定义为比任何其他字符都小（更多信息）。


提示：
1 <= words.length <= 100
1 <= words[i].length <= 20
order.length == 26
在 words[i] 和 order 中的所有字符都是英文小写字母。"""


class NotSortedException(Exception):
    def __init__(self, *args: object) -> None:
        super().__init__(*args)


class Solution:
    def isAlienSorted(self, words, order):
        orderd = {}
        for i, a in zip(range(len(order)), order):
            orderd[a] = i
        
        is_sorted = True
        try:
            for i in range(len(words)-1):
                word, follow = words[i], words[i+1]
                j = 0
                while True:
                    try:
                        a = word[j]
                    except IndexError:
                        break

                    b = follow[j]

                    if orderd[a] < orderd[b]:
                        break
                    elif orderd[a] == orderd[b]:
                        j += 1
                        continue
                    else:
                        raise NotSortedException(a, b)
        except (IndexError, NotSortedException):
            is_sorted = False
        
        return is_sorted
                

if __name__ == '__main__':
    so = Solution()
    print(so.isAlienSorted(words=["hello","leetcode"], order="hlabcdefgijkmnopqrstuvwxyz"))
    print(so.isAlienSorted(words=["word","world","row"], order="worldabcefghijkmnpqstuvxyz"))
    print(so.isAlienSorted(words=["apple","app"], order="abcdefghijklmnopqrstuvwxyz"))
